package com.study.lib_tiandi_map.utils

import kotlin.math.*

object TransferOfAxesUtil {

    //定义一些常量
    private const val x_PI = 3.14159265358979324 * 3000.0 / 180.0
    private const val PI = 3.1415926535897932384626
    private const val a = 6378245.0
    private const val ee = 0.00669342162296594323

    /**
     * WGS84转GCj02
     * @param lng
     * @param lat
     * @returns {*[]}
     */
    fun wgs84ToGcj02(lng: Double, lat: Double): List<Double> {
        return if (outOfChina(lng, lat)) {
            listOf(lng, lat)
        } else {
            var dlat = transformLat(lng - 105.0, lat - 35.0)
            var dlng = transformLng(lng - 105.0, lat - 35.0)
            val radLat = lat / 180.0 * PI
            var magic = sin(radLat)
            magic = 1 - ee * magic * magic
            val sqrtMagic = sqrt(magic);
            dlat = (dlat * 180.0) / ((a * (1 - ee)) / (magic * sqrtMagic) * PI)
            dlng = (dlng * 180.0) / (a / sqrtMagic * cos(radLat) * PI)
            val mgLat = lat + dlat
            val mgLng = lng + dlng
            listOf(mgLng, mgLat)
        }
    }

    /**
     * GCJ02 转换为 WGS84
     * @param lng
     * @param lat
     * @returns {*[]}
     */
    fun gcj02ToWgs84(lng: Double, lat: Double): List<Double> {
        return if (outOfChina(lng, lat)) {
            listOf(lng, lat)
        } else {
            var dlat = transformLat(lng - 105.0, lat - 35.0)
            var dlng = transformLng(lng - 105.0, lat - 35.0)
            val radLat = lat / 180.0 * PI
            var magic = sin(radLat)
            magic = 1 - ee * magic * magic
            val sqrtMagic = sqrt(magic)
            dlat = (dlat * 180.0) / ((a * (1 - ee)) / (magic * sqrtMagic) * PI)
            dlng = (dlng * 180.0) / (a / sqrtMagic * cos(radLat) * PI)
            val mgLat = lat + dlat
            val mgLng = lng + dlng
            listOf(lng * 2 - mgLng, lat * 2 - mgLat)
        }
    }

    private fun transformLat(lng: Double, lat: Double): Double {
        var ret =
            -100.0 + 2.0 * lng + 3.0 * lat + 0.2 * lat * lat + 0.1 * lng * lat + 0.2 * sqrt(
                abs(lng)
            );
        ret += (20.0 * sin(6.0 * lng * PI) + 20.0 * sin(2.0 * lng * PI)) * 2.0 / 3.0
        ret += (20.0 * sin(lat * PI) + 40.0 * sin(lat / 3.0 * PI)) * 2.0 / 3.0
        ret += (160.0 * sin(lat / 12.0 * PI) + 320 * sin(lat * PI / 30.0)) * 2.0 / 3.0
        return ret
    }

    private fun transformLng(lng: Double, lat: Double): Double {
        var ret = 300.0 + lng + 2.0 * lat + 0.1 * lng * lng + 0.1 * lng * lat + 0.1 * sqrt(
            abs(lng)
        );
        ret += (20.0 * sin(6.0 * lng * PI) + 20.0 * sin(2.0 * lng * PI)) * 2.0 / 3.0
        ret += (20.0 * sin(lng * PI) + 40.0 * sin(lng / 3.0 * PI)) * 2.0 / 3.0
        ret += (150.0 * sin(lng / 12.0 * PI) + 300.0 * sin(lng / 30.0 * PI)) * 2.0 / 3.0
        return ret
    }

    /**
     * 判断是否在国内，不在国内则不做偏移
     * @param lng
     * @param lat
     * @returns {boolean}
     */
    private fun outOfChina(lng: Double, lat: Double): Boolean {
        return (lng < 72.004 || lng > 137.8347) || (lat < 0.8293 || lat > 55.8271)
    }

}